EfficientNetB1

class EfficientNetB1(noTop: Boolean) : ONNXModels.CV
Content copied to clipboard

This model is a neural network for image classification that take images as input and classify the major object in the image into a set of 1000 different classes (labels are available via org.jetbrains.kotlinx.dl.impl.dataset.Imagenet.V1k.labels method).

Instantiates the EfficientNetB1 architecture.

The model have

  • an input with the shape (1x240x240x3)

  • an output with the shape (1x1000)

  • an output for noTop model with the shape (1x7x7x1280)

NOTE: This model is converted from Keras.applications, the last two layers in the noTop model have been removed so that the user can fine-tune the model for his specific task.

See also

Constructors

EfficientNetB1
Link copied to clipboard
fun EfficientNetB1(noTop: Boolean = false)
Content copied to clipboard

Functions

model
Link copied to clipboard
open fun model(modelHub: ModelHub): OnnxInferenceModel
Content copied to clipboard
pretrainedModel
Link copied to clipboard
open override fun pretrainedModel(modelHub: ModelHub): ImageRecognitionModel
Content copied to clipboard

Properties

inputShape
Link copied to clipboard
open val inputShape: LongArray?
Content copied to clipboard

Shape of the input accepted by this model, without batch size.

modelRelativePath
Link copied to clipboard
open override val modelRelativePath: String
Content copied to clipboard
preprocessor
Link copied to clipboard
open val preprocessor: Operation<Pair<FloatArray, TensorShape>, Pair<FloatArray, TensorShape>>
Content copied to clipboard